package com.wx.user.controller;



import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.api.ApiController;
import com.baomidou.mybatisplus.extension.api.R;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.wx.commons.entity.Role;
import com.wx.commons.entity.UserRole;
import com.wx.commons.model.RoleSaveReq;
import com.wx.user.service.RoleService;
import com.wx.user.service.UserRoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.io.Serializable;
import java.util.List;
import java.util.stream.Collectors;

/**
  * 角色表(role)表控制层
  * @author liuwenxuan
  * @TableName role
  * @createDate 20240627.01
  */
@RestController
@RequestMapping("role")
public class RoleController extends ApiController {
    /**
     * 服务对象
     */
    @Resource
    private RoleService roleService;

    @Autowired
    private UserRoleService userRoleService;

    /**
     * 分页查询所有数据
     *
     * @param page 分页对象
     * @param role 查询实体
     * @return 所有数据
     */
    @GetMapping
    public R selectAll(Page<Role> page, Role role) {
        return success(this.roleService.page(page, new QueryWrapper<>(role)));
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public R selectOne(@PathVariable Serializable id) {
        return success(this.roleService.getById(id));
    }

    /**
     * 新增数据
     *
     * @param role 实体对象
     * @return 新增结果
     */
    @PostMapping
    public R insert(@RequestBody Role role) {
        return success(this.roleService.save(role));
    }

    /**
     * 修改数据
     *
     * @param role 实体对象
     * @return 修改结果
     */
    @PutMapping
    public R update(@RequestBody Role role) {
        return success(this.roleService.updateById(role));
    }

    /**
     * 删除数据
     *
     * @param idList 主键结合
     * @return 删除结果
     */
    @DeleteMapping
    public R delete(@RequestParam("idList") List<Long> idList) {
        return success(this.roleService.removeByIds(idList));
    }

    @PostMapping("/batch")
    @Transactional(rollbackFor = Exception.class)
    public R batch(@RequestBody RoleSaveReq roleSaveReq) {
        if (CollectionUtil.isNotEmpty(roleSaveReq.getDel())){
            List<Long> roleIds = roleSaveReq.getDel().stream().map(Role::getId).collect(Collectors.toList());
            this.roleService.removeByIds(roleIds);
        }
        if (CollectionUtil.isNotEmpty(roleSaveReq.getInsertAndUpdate())){
            this.roleService.saveOrUpdateBatch(roleSaveReq.getInsertAndUpdate());
        }
        return success(true);
    }

    @GetMapping("/getCurRole")
    public R getCurRole(@RequestHeader("user")String user) {
        // 解析json格式字符串user
        JSONObject jsonObject = JSONUtil.parseObj(user);
        Long userId = jsonObject.getLong("id");
        List<Long> roleIds = userRoleService.list(new QueryWrapper<UserRole>().eq("user_id", userId))
                .stream().map(UserRole::getRoleId)
                .collect(Collectors.toList());
        return success(roleService.listByIds(roleIds).stream().map(Role::getName));
    }

}

